from calculator import Count


# 测试相加
class TestCount:
    def test_add(self):
        try:
            j = Count(3, 3)
            add = j.add()
            assert (add == 5), 'Integer addition result error'
        except AssertionError as msg:
            print(msg)
        else:
            print('Test pass')


mytest = TestCount()
mytest.test_add()
